/**
 * 小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为 root 。
 *
 * 除了 root 之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果 两个直接相连的房子在同一天晚上被打劫 ，房屋将自动报警。
 *
 * 给定二叉树的 root 。返回 在不触动警报的情况下 ，小偷能够盗取的最高金额 。
 *
 *https://leetcode.cn/problems/house-robber-iii/description/
 * 题解：https://labuladong.gitee.io/algo/3/28/95/
 */
class RobThree {
    Map<TreeNode,Integer> memo=new HashMap<>();
    public int robDp(TreeNode root) {
        if(root==null) {
            return 0;
        }
        if(memo.containsKey(root)) {
            return memo.get(root);
        }
        int do_it=root.val+(root.left==null?0:rob(root.left.left)+rob(root.left.right))+(root.right==null?0:rob(root.right.left)+rob(root.right.right));
        int not_do=rob(root.left)+rob(root.right);
        int res=Math.max(do_it,not_do);
        memo.put(root,res);
        return res;
    }
    public int rob(TreeNode root) {
        int[] res=dp(root);
        return Math.max(res[0],res[1]);
    }
    public int[] dp(TreeNode root) {
        if(root==null) {
            return new int[]{0,0};
        }
        /**
        返回一个大小为2的arr数组，arr[0]表示不抢root的话，得到的最大钱数 ；arr[1]表示抢的话，得到的最大钱数*/
        int[] left=dp(root.left);
        int[] right=dp(root.right);
        //抢，下家就不能抢了
        int rob=root.val+left[0]+right[0];
        //不抢，下家可抢可不抢，取决于收益大小
        int not_rob=Math.max(left[0],left[1])+Math.max(right[0],right[1]);
        return new int[]{not_rob,rob};
    }





}